588D - Duff in Beach - CodeForces Solution


*2100

Please click on ads to support us..

Python Code:

from collections import Counter
mod = 10 ** 9 + 7
n, l, k = list(map(int, input().split()))
if k == 1:
    print(l % mod)
    exit()
a = list(map(int, input().split()))
if l < n:
    a = a[:l]
    n = len(a)
m = {v : i + 1 for i, v in enumerate(sorted(set(a)))}
b = [m[i] for i in a]
g = l // n
r = l % n
k = min(k, g + int(r > 0))
d = Counter(b)
dr = Counter(b[:r])
sd = sorted(d.items())

dp = [0] * (len(m) + 1)
dp[0] = 1
ret = 0
for kk in range(k):
    tmp = [0] * len(dp)
    psum = dp[0]
    for v, c in sd:
        psum = (psum + dp[v]) % mod
        tmp[v] = psum * c % mod
        rep = (g - kk) % mod * c % mod + dr.get(v, 0)
                ret = (ret + psum * rep % mod) % mod
    dp = tmp
print(ret)


Comments

Submit
0 Comments
More Questions

1716B - Permutation Chain
987A - Infinity Gauntlet
1676G - White-Black Balanced Subtrees
1716D - Chip Move
1352F - Binary String Reconstruction
1487B - Cat Cycle
1679C - Rooks Defenders
56A - Bar
1694B - Paranoid String
35A - Shell Game
1684A - Digit Minimization
43B - Letter
1017A - The Rank
1698B - Rising Sand
235A - LCM Challenge
1075B - Taxi drivers and Lyft
1562A - The Miracle and the Sleeper
1216A - Prefixes
1490C - Sum of Cubes
868A - Bark to Unlock
873B - Balanced Substring
1401D - Maximum Distributed Tree
1716C - Robot in a Hallway
1688B - Patchouli's Magical Talisman
99A - Help Far Away Kingdom
622B - The Time
1688C - Manipulating History
1169D - Good Triple
1675B - Make It Increasing
588A - Duff and Meat